from mgamdata.dataset.RenJi_Sarcopenia.mm_dataset import Sarcopenia_Mha, Sarcopenia_Precrop_Npz
from mgamdata.dataset.Totalsegmentator.mm_dataset import TotalsegmentatorSeg3DDataset, Tsd3D_PreCrop_Npz
from mgamdata.dataset.FLARE_2023.mm_dataset import FLARE_2023_Semi_Mha, FLARE_2023_Precrop_Npz
from mgamdata.dataset.AbdomenCT_1K.mm_dataset import AbdomenCT_1K_Semi_Mha, AbdomenCT_1K_Precrop_Npz
from mgamdata.dataset.KiTS23.mm_dataset import KiTS23_Mha, KiTS23_Precrop_Npz
from mgamdata.dataset.LUNA16.mm_dataset import LUNA16_Mha, LUNA16_Precrop_Npz


Sar_mha_data_root           = "/home/zhangyq.sx/mgam_datasets/Sarcopenia_Data/spacing2_mha/"
Tsd_mha_data_root           = "/home/zhangyq.sx/mgam_datasets/Totalsegmentator_Data/spacing2_mha/"
FLARE_mha_data_root         = "/home/zhangyq.sx/mgam_datasets/FLARE_2023_Data/spacing2_mha/"
KiTS23_mha_data_root        = "/home/zhangyq.sx/mgam_datasets/KiTS23_Data/spacing2_mha/"
AbCT1K_mha_data_root        = "/home/zhangyq.sx/mgam_datasets/AbdomenCT1K_Data/spacing2_mha/"
LUNA16_mha_data_root        = "/home/zhangyq.sx/mgam_datasets/LUNA16_Data/spacing2_mha/"

Sar_crop96_data_root        = "/home/zhangyq.sx/mgam_datasets/Sarcopenia_Data/spacing2_crop96_npz/"
Tsd_crop96_data_root        = "/home/zhangyq.sx/mgam_datasets/Totalsegmentator_Data/spacing2_crop96_npz/"
FLARE_crop96_data_root      = "/home/zhangyq.sx/mgam_datasets/FLARE_2023_Data/spacing2_crop96_npz/"
KiTS23_crop96_data_root     = "/home/zhangyq.sx/mgam_datasets/KiTS23_Data/spacing2_crop96_npz/"
AbCT1K_crop96_data_root     = "/home/zhangyq.sx/mgam_datasets/AbdomenCT1K_Data/spacing2_crop96_npz/"
LUNA16_crop96_data_root     = "/home/zhangyq.sx/mgam_datasets/LUNA16_Data/spacing2_crop96_npz/"

Sar_crop48_data_root        = "/home/zhangyq.sx/mgam_datasets/Sarcopenia_Data/spacing2_crop48_npz/"
Tsd_crop48_data_root        = "/home/zhangyq.sx/mgam_datasets/Totalsegmentator_Data/spacing2_crop48_npz/"
FLARE_crop48_data_root      = "/home/zhangyq.sx/mgam_datasets/FLARE_2023_Data/spacing2_crop48_npz/"
KiTS23_crop48_data_root     = "/home/zhangyq.sx/mgam_datasets/KiTS23_Data/spacing2_crop48_npz/"
AbCT1K_crop48_data_root     = "/home/zhangyq.sx/mgam_datasets/AbdomenCT1K_Data/spacing2_crop48_npz/"
LUNA16_crop48_data_root     = "/home/zhangyq.sx/mgam_datasets/LUNA16_Data/spacing2_crop48_npz/"


TsDataset = dict(
    type=TotalsegmentatorSeg3DDataset,
    data_root=Tsd_mha_data_root,
    data_root_mha=Tsd_mha_data_root,
)
FLARE2023Dataset = dict(
    type=FLARE_2023_Semi_Mha,
    data_root=FLARE_mha_data_root,
    data_root_mha=FLARE_mha_data_root,
)
SarcopeniaDataset = dict(
    type=Sarcopenia_Mha, 
    data_root=Sar_mha_data_root,
    data_root_mha=Sar_mha_data_root,
)
KiTS23_Dataset = dict(
    type=KiTS23_Mha,
    data_root=KiTS23_mha_data_root,
    data_root_mha=KiTS23_mha_data_root,
)
AbdomenCT1K_Dataset = dict(
    type=AbdomenCT_1K_Semi_Mha,
    data_root=AbCT1K_mha_data_root,
    data_root_mha=AbCT1K_mha_data_root,
)
LUNA16_Dataset = dict(
    type=LUNA16_Mha,
    data_root=LUNA16_mha_data_root,
    data_root_mha=LUNA16_mha_data_root,
)

Ts_Crop96 = dict(
    type=Tsd3D_PreCrop_Npz,
    data_root=Tsd_crop96_data_root,
    data_root_mha=Tsd_mha_data_root,
)
FLARE23_Crop96 = dict(
    type=FLARE_2023_Precrop_Npz,
    data_root=FLARE_crop96_data_root,
    data_root_mha=FLARE_mha_data_root,
)
Sar_Crop96 = dict(
    type=Sarcopenia_Precrop_Npz,
    data_root=Sar_crop96_data_root,
    data_root_mha=Sar_mha_data_root,
)
KiTS23_Crop96 = dict(
    type=KiTS23_Precrop_Npz,
    data_root=KiTS23_crop96_data_root,
    data_root_mha=KiTS23_mha_data_root,
)
AbCT1K_Crop96 = dict(
    type=AbdomenCT_1K_Precrop_Npz,
    data_root=AbCT1K_crop96_data_root,
    data_root_mha=AbCT1K_mha_data_root,
)
LUNA16_Crop96 = dict(
    type=LUNA16_Precrop_Npz,
    data_root=LUNA16_crop96_data_root,
    data_root_mha=LUNA16_mha_data_root,
)

Ts_Crop48 = dict(
    type=Tsd3D_PreCrop_Npz,
    data_root=Tsd_crop48_data_root,
    data_root_mha=Tsd_mha_data_root,
)
FLARE23_Crop48 = dict(
    type=FLARE_2023_Precrop_Npz,
    data_root=FLARE_crop48_data_root,
    data_root_mha=FLARE_mha_data_root,
)
Sar_Crop48 = dict(
    type=Sarcopenia_Precrop_Npz,
    data_root=Sar_crop48_data_root,
    data_root_mha=Sar_mha_data_root,
)
KiTS23_Crop48 = dict(
    type=KiTS23_Precrop_Npz,
    data_root=KiTS23_crop48_data_root,
    data_root_mha=KiTS23_mha_data_root,
)
AbCT1K_Crop48 = dict(
    type=AbdomenCT_1K_Precrop_Npz,
    data_root=AbCT1K_crop48_data_root,
    data_root_mha=AbCT1K_mha_data_root,
)
LUNA16_Crop48 = dict(
    type=LUNA16_Precrop_Npz,
    data_root=LUNA16_crop48_data_root,
    data_root_mha=LUNA16_mha_data_root,
)
